package cn.iocoder.yudao.module.biz.service.risk.taskWork;

import cn.hutool.core.collection.CollUtil;
import cn.iocoder.yudao.framework.mybatis.core.query.QueryWrapperX;
import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
import cn.iocoder.yudao.module.biz.controller.admin.risk.basicManage.vo.BizBasicManageCreateReqVO;
import cn.iocoder.yudao.module.biz.controller.admin.risk.basicManage.vo.BizBasicManageExcelVO;
import cn.iocoder.yudao.module.biz.controller.admin.risk.basicManage.vo.BizBasicManageImportRespVO;
import cn.iocoder.yudao.module.biz.controller.admin.risk.taskWork.vo.*;
import cn.iocoder.yudao.module.biz.convert.risk.taskWork.BizTaskWorkConvert;
import cn.iocoder.yudao.module.biz.dal.dataobject.risk.taskWork.BizTaskWorkDO;
import cn.iocoder.yudao.module.biz.dal.mysql.risk.taskWork.BizTaskWorkMapper;
import cn.iocoder.yudao.module.system.dal.dataobject.dept.DeptDO;
import cn.iocoder.yudao.module.system.dal.mysql.dept.DeptMapper;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;

import org.springframework.transaction.annotation.Transactional;
import org.springframework.validation.annotation.Validated;
import java.util.*;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.biz.enums.ErrorCodeConstants.*;
import static cn.iocoder.yudao.module.system.enums.ErrorCodeConstants.USER_IMPORT_LIST_IS_EMPTY;

/**
 * 作业活动 Service 实现类
 *
 * @author 芋道源码
 */
@Service
@Validated
public class BizTaskWorkServiceImpl implements BizTaskWorkService {

    @Resource
    private BizTaskWorkMapper taskWorkMapper;

    @Resource
    private DeptMapper deptMapper;

    @Override
    public String createTaskWork(BizTaskWorkCreateReqVO createReqVO) {
        // 插入
        BizTaskWorkDO taskWork = BizTaskWorkConvert.INSTANCE.convert(createReqVO);
        taskWork.setDeptId(String.valueOf(SecurityFrameworkUtils.getLoginUser().getDeptId()));
        taskWorkMapper.insert(taskWork);
        // 返回
        return taskWork.getId();
    }

    @Override
    public void updateTaskWork(BizTaskWorkUpdateReqVO updateReqVO) {
        // 校验存在
        this.validateTaskWorkExists(updateReqVO.getId());
        // 更新
        BizTaskWorkDO updateObj = BizTaskWorkConvert.INSTANCE.convert(updateReqVO);
        taskWorkMapper.updateById(updateObj);
    }

    @Override
    public void deleteTaskWork(String id) {
        // 校验存在
        this.validateTaskWorkExists(id);
        // 删除
        taskWorkMapper.deleteById(id);
    }

    private void validateTaskWorkExists(String id) {
        if (taskWorkMapper.selectById(id) == null) {
            throw exception(TASK_WORK_NOT_EXISTS);
        }
    }

    @Override
    public BizTaskWorkDO getTaskWork(String id) {
        return taskWorkMapper.selectById(id);
    }

    @Override
    public List<BizTaskWorkDO> getTaskWorkList(Collection<String> ids) {
        return taskWorkMapper.selectBatchIds(ids);
    }

    @Override
    public PageResult<BizTaskWorkDO> getTaskWorkPage(BizTaskWorkPageReqVO pageReqVO) {
        pageReqVO.setDeptId(String.valueOf(SecurityFrameworkUtils.getLoginUser().getDeptId()));
        return taskWorkMapper.selectPage(pageReqVO);
    }

    @Override
    public List<BizTaskWorkDO> getTaskWorkList(BizTaskWorkExportReqVO exportReqVO) {
        exportReqVO.setDeptId(String.valueOf(SecurityFrameworkUtils.getLoginUser().getDeptId()));
        return taskWorkMapper.selectList(exportReqVO);
    }

    @Override
    public List<BizTaskWorkRespVO> getTaskPage(BizTaskWorkPageReqVO pageReqVO) {
        Integer pageNo=pageReqVO.getPageNo();
        Integer pageSize=  pageReqVO.getPageSize();
        //设置为limit开始的条数
        pageReqVO.setPageNo((pageNo-1)*pageSize);
        pageReqVO.setDeptId(SecurityFrameworkUtils.getLoginUser().getDeptId()+"");
        return taskWorkMapper.getTaskPage(pageReqVO);
    }

    @Override
    public Integer getTaskCount(BizTaskWorkPageReqVO pageReqVO) {
        return taskWorkMapper.getTaskCount(pageReqVO);
    }

    @Override
    @Transactional(rollbackFor = Exception.class) // 添加事务，异常则回滚所有导入
    public BizTaskWorkImportRespVO importExcel(List<BizTaskWorkExcelVO> importTask) {
        if (CollUtil.isEmpty(importTask)) {
            throw exception(USER_IMPORT_LIST_IS_EMPTY);
        }
        BizTaskWorkImportRespVO respVO = BizTaskWorkImportRespVO.builder().failureTasknames(new LinkedHashMap<>()).build();
        Map<String,String> map = new HashMap<>();
        for(BizTaskWorkExcelVO vo : importTask){
            //判断实施单位是否存在
            DeptDO deptDO = deptMapper.selectOne(new QueryWrapperX<DeptDO>().eqIfPresent("name",vo.getExplotUnit()));
            if(deptDO == null){//如果实施单位不存在
                map.put(vo.getActivityName()+"的实施单位","不存在，请手动修改");
                vo.setExplotUnit(null);
                respVO.setFailureTasknames(map);
            }else{
                vo.setExplotUnit(String.valueOf(deptDO.getId()));
            }
            BizTaskWorkCreateReqVO taskWorkCreateReqVO = new BizTaskWorkCreateReqVO();
            taskWorkCreateReqVO.setActivityName(vo.getActivityName());
            taskWorkCreateReqVO.setActivityContent(vo.getActivityContent());
            taskWorkCreateReqVO.setWorkArea(vo.getWorkArea());
            taskWorkCreateReqVO.setIsSpecial(vo.getIsSpecial());
            taskWorkCreateReqVO.setExplotUnit(vo.getExplotUnit());
            taskWorkCreateReqVO.setActivityFrequency(vo.getActivityFrequency());
            taskWorkCreateReqVO.setWorkerCount(vo.getWorkerCount());
            taskWorkCreateReqVO.setRemarks(vo.getRemarks());
            createTaskWork(taskWorkCreateReqVO);
        }
        return respVO;
    }

    @Override
    public List<BizTaskWorkDO> getTaskForExcel(BizTaskWorkExportReqVO exportReqVO) {
        exportReqVO.setDeptId(String.valueOf(SecurityFrameworkUtils.getLoginUser().getDeptId()));
        return taskWorkMapper.getTaskForExcel(exportReqVO);
    }

}
